package com.ruoyi.framework.web.domain;

import com.ruoyi.framework.web.domain.server.Process;
import com.ruoyi.framework.web.domain.server.*;
import oshi.SystemInfo;
import oshi.hardware.CentralProcessor;
import oshi.hardware.CentralProcessor.TickType;
import oshi.hardware.GlobalMemory;
import oshi.hardware.HardwareAbstractionLayer;
import oshi.hardware.NetworkIF;
import oshi.software.os.NetworkParams;
import oshi.software.os.OSProcess;
import oshi.software.os.OperatingSystem;
import oshi.util.FormatUtil;
import oshi.util.Util;

import java.math.BigDecimal;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Properties;

/**
 * 服务器相关信息
 */
public class Server
{
    private static final int OSHI_WAIT_SECOND = 1000;
    
    /**
     * CPU相关信息
     */
    private Cpu cpu = new Cpu();

    /**
     * 內存相关信息
     */
    private Mem mem = new Mem();

    /**
     * JVM相关信息
     */
    private Jvm jvm = new Jvm();

    /**
     * 服务器进程相关信息
     */
    private ArrayList<Process> process =new ArrayList<>();

    /**
     * 服务器全部网络接口
     */
    private ArrayList<networkInterface> networkInterfaces= new ArrayList<>();

    /**
     * 网络相关信息
     */
    private Network network=new Network();


    public Cpu getCpu()
    {
        return cpu;
    }

    public void setCpu(Cpu cpu)
    {
        this.cpu = cpu;
    }

    public Mem getMem()
    {
        return mem;
    }

    public void setMem(Mem mem)
    {
        this.mem = mem;
    }

    public Jvm getJvm()
    {
        return jvm;
    }

    public void setJvm(Jvm jvm)
    {
        this.jvm = jvm;
    }

    public Network getNetwork() {
        return network;
    }

    public void setNetwork(Network network) {
        this.network = network;
    }


    public ArrayList<Process> getProcess() {
        return process;
    }

    public void setProcess(ArrayList<Process> process) {
        this.process = process;
    }

    public ArrayList<networkInterface> getNetworkInterfaces() {
        return networkInterfaces;
    }

    public void setNetworkInterfaces(ArrayList<networkInterface> networkInterfaces) {
        this.networkInterfaces = networkInterfaces;
    }
    public void copyTo() throws Exception
    {
        SystemInfo si = new SystemInfo();
        HardwareAbstractionLayer hal = si.getHardware();
        OperatingSystem os = si.getOperatingSystem();
        setCpuInfo(hal.getProcessor());
        setMemInfo(hal.getMemory());
        setProcess(hal,os);
        setNetworkInfo(si.getOperatingSystem());
        setNetworkInterface(hal);
        setJvmInfo();

    }
    private void setNetworkInterface(HardwareAbstractionLayer hal){
        List<NetworkIF> networkIFs = hal.getNetworkIFs();
        for (NetworkIF net : networkIFs) {
            networkInterface networkInterface =new networkInterface();
            networkInterface.setName(net.getName());
            networkInterface.setMACAddress(net.getMacaddr());
            networkInterface.setMTU(net.getMTU());
            networkInterface.setSpeed(FormatUtil.formatValue(net.getSpeed(), "bps"));
            networkInterface.setIPv4( Arrays.toString(net.getIPv4addr()));
            networkInterface.setIPv6(Arrays.toString(net.getIPv6addr()));

            boolean hasData = net.getBytesRecv() > 0 || net.getBytesSent() > 0 || net.getPacketsRecv() > 0
                    || net.getPacketsSent() > 0;
            if (hasData){
                networkInterface.setReceived(String.valueOf(net.getPacketsRecv())+" packets");
                networkInterface.setReceivedPackets(FormatUtil.formatBytes(net.getBytesRecv()));
                networkInterface.setReceivedErr(net.getInErrors() + " err");
                networkInterface.setTransmitted(net.getPacketsSent() + " packets");
                networkInterface.setTransmittedPackets(FormatUtil.formatBytes(net.getBytesSent()));
                networkInterface.setTransmittedPacketsErr(net.getOutErrors() + " err");
            }
        networkInterfaces.add(networkInterface);
        }
    }
    private void setProcess(HardwareAbstractionLayer hal,OperatingSystem os){

        List<OSProcess> procs = os.getProcesses(5, OperatingSystem.ProcessSort.CPU);
        for (int i = 0; i < procs.size() && i < 5; i++) {
            Process proc=new Process();
            OSProcess p = procs.get(i);
            double f=100d * (p.getKernelTime() + p.getUserTime()) / p.getUpTime();
            BigDecimal b = new BigDecimal(f);
            double nf = b.setScale(2,   BigDecimal.ROUND_HALF_UP).doubleValue();
            proc.setCPU(String.valueOf(nf));
            proc.setPID(String.valueOf(p.getProcessID()));
            proc.setMEM(String.valueOf(100d * p.getResidentSetSize() / hal.getMemory().getTotal()));
            proc.setRSS(String.valueOf(FormatUtil.formatBytes(p.getResidentSetSize())));
            proc.setName(p.getName());
            proc.setVSZ(String.valueOf(FormatUtil.formatBytes(p.getVirtualSize())));
            process.add(proc);

        }
    }
    /**
     * 设置CPU信息
     */
    private void setCpuInfo(CentralProcessor processor)
    {
        // CPU信息
        long[] prevTicks = processor.getSystemCpuLoadTicks();
        Util.sleep(OSHI_WAIT_SECOND);
        long[] ticks = processor.getSystemCpuLoadTicks();
        long nice = ticks[TickType.NICE.getIndex()] - prevTicks[TickType.NICE.getIndex()];
        long irq = ticks[TickType.IRQ.getIndex()] - prevTicks[TickType.IRQ.getIndex()];
        long softirq = ticks[TickType.SOFTIRQ.getIndex()] - prevTicks[TickType.SOFTIRQ.getIndex()];
        long steal = ticks[TickType.STEAL.getIndex()] - prevTicks[TickType.STEAL.getIndex()];
        long cSys = ticks[TickType.SYSTEM.getIndex()] - prevTicks[TickType.SYSTEM.getIndex()];
        long user = ticks[TickType.USER.getIndex()] - prevTicks[TickType.USER.getIndex()];
        long iowait = ticks[TickType.IOWAIT.getIndex()] - prevTicks[TickType.IOWAIT.getIndex()];
        long idle = ticks[TickType.IDLE.getIndex()] - prevTicks[TickType.IDLE.getIndex()];
        long totalCpu = user + nice + cSys + idle + iowait + irq + softirq + steal;
        cpu.setCpuNum(processor.getLogicalProcessorCount());
        cpu.setTotal(totalCpu);
        cpu.setSys(cSys);
        cpu.setUsed(user);
        cpu.setWait(iowait);
        cpu.setFree(idle);
    }

    /**
     * 设置内存信息
     */
    private void setMemInfo(GlobalMemory memory)
    {
        mem.setTotal(memory.getTotal());
        mem.setUsed(memory.getTotal() - memory.getAvailable());
        mem.setFree(memory.getAvailable());
    }


    /**
     * 设置Java虚拟机
     */
    private void setJvmInfo() throws UnknownHostException
    {
        Properties props = System.getProperties();
        jvm.setTotal(Runtime.getRuntime().totalMemory());
        jvm.setMax(Runtime.getRuntime().maxMemory());
        jvm.setFree(Runtime.getRuntime().freeMemory());
        jvm.setVersion(props.getProperty("java.version"));
        jvm.setHome(props.getProperty("java.home"));
    }

    /**
     * 设置网络环境
     * @param os
     */
    private void setNetworkInfo(OperatingSystem os){

        NetworkParams networkParams = os.getNetworkParams();
        network.setHostName(networkParams.getHostName());
        network.setDNSServers(Arrays.toString(networkParams.getDnsServers()));
        network.setDomainName(networkParams.getDomainName());
        network.setIPv4(networkParams.getIpv4DefaultGateway());
        network.setIPv6(networkParams.getIpv6DefaultGateway());
    }


    /**
     * 字节转换
     * 
     * @param size 字节大小
     * @return 转换后值
     */
    public String convertFileSize(long size)
    {
        long kb = 1024;
        long mb = kb * 1024;
        long gb = mb * 1024;
        if (size >= gb)
        {
            return String.format("%.1f GB", (float) size / gb);
        }
        else if (size >= mb)
        {
            float f = (float) size / mb;
            return String.format(f > 100 ? "%.0f MB" : "%.1f MB", f);
        }
        else if (size >= kb)
        {
            float f = (float) size / kb;
            return String.format(f > 100 ? "%.0f KB" : "%.1f KB", f);
        }
        else
        {
            return String.format("%d B", size);
        }
    }


}
